|
wallpaper engine可以用本地视频作为桌面背景，如果电脑里有很多电影、电视剧、动画之类的想随机播放，一个一个添加还是很困难的，本文用python写一个脚本，修改wallpaper engine的配置文件，把指定目录下的视频全部添加进来。算是学习python的一个小例子吧。
#-*- coding: utf-8 -*-
import json
import os
import io
import re
#输入一个目录，返回目录下所有的视频文件（包括子目录）
def getFileList(file_dir):
L = []
#筛选用的扩展名，列出wallpaper engine支持的视频文件类型
extNames=['.mp4','.wmv','.avi','.m4v','.mov','.webm','.mkv']
for root, dirs, files in os.walk(file_dir):
for file in files:
extName=os.path.splitext(file)[1]
if extName in extNames:
L.append(os.path.join(root, file))
return L
#默认的wallpaper engine配置文件路径
filename="C:\Program Files (x86)\Steam\steamapps\common\wallpaper_engine\config.json"
#用with而不是直接f=open(filename,'r')，可以保证文件打开失败程序继续运行
#然而对这个程序来说没什么用
with open(filename,'r') as f:
#按utf-8编码打开json
data=json.load(f,encoding="utf-8")
#找到localfiles元素，这里的Noah是这台电脑的用户名
#在修改wallpaper engine的设置时，经常出现没有用户名而直接跳到general的情况
#我也不知道为什么，所以这里应该根据实际情况修改
listOri=data["Noah"]["general"]["localfiles"]
#将本地视频列表与已有的列表合并
listNew=getFileList(u'D:\Download\视频')+listOri
#去掉重复的文件路径
listN=[]
for li in listNew:
if li not in listN:
listN.append(li)
#将列表赋值给json元素
data["Noah"]["general"]["localfiles"]=listN
#这里打开文件与上面不同，因为默认的open不能设置编码格式
with io.open(filename,'w', encoding='utf-8') as f:
#ensure_ascii=False：json.dump默认忽略特殊字符
#indent=4：子元素缩进四格
str=json.dumps(data, ensure_ascii=False,indent=4)
#将缩进替换为‘\t’，这是python3.2之前的写法，之后可以直接设置indent='\t'
#从别处抄来的，没看懂
str = re.sub('\n +', lambda match: '\n' + '\t' * (len(match.group().strip('\n')) / 3), str)
f.write(str)
|